package service

import (
	"monitor_serve/initialize"
	"monitor_serve/model"
)

type UserTokenService struct {
}

// UserTokenSave 保存用户token。查看用户是否存在，存在则更新，不存在则插入
func (a *UserTokenService) UserTokenSave(username, token string) error {
	userToken := model.UserToken{
		Username: username, // 确保 UserToken 结构体中有 Username 字段
		Token:    token,
	}
	return initialize.DB.Where("username = ?", username).Assign(userToken).FirstOrCreate(&model.UserToken{}).Error
}
